package middleware

import (
	"github.com/gin-gonic/gin"
	"net/http"
)

// cors 允许跨域
func cors() gin.HandlerFunc {
	return func(context *gin.Context) {
		method := context.Request.Method
		context.Header("Access-Control-Allow-Origin", "*")
		context.Header("Access-Control-Allow-Headers", "*")
		context.Header("Access-Control-Allow-Methods", "POST, GET, OPTIONS")
		context.Header("Access-Control-Allow-Credentials", "true")
		if method == "OPTIONS" {
			context.AbortWithStatus(http.StatusNoContent)
			context.Abort()
		} else {
			context.Next()
		}
	}
}
func UseCors(router *gin.Engine) {
	router.Use(cors())
}
